# -*- coding: utf-8 -*-
"""
Module Description: 数据层管理中心
"""

# 设置dal.orm用来获取当前访问的db名称的方法
# 便于测试时灵活指定
_set_db_func = None


def set_db_func(f):
    """
    用于指定数据库名称
    :param f: 返回值为DBNAME的函数
    """
    global _set_db_func
    _set_db_func = f


def get_db_name():
    """返回当前连接的db名称"""
    if _set_db_func:
        return _set_db_func()